#!/bin/bash
echo "==============================="
echo "🚀 FastAPI Dog Breed Trainer 启动脚本 (Linux/macOS)"
echo "==============================="

# Step 1: 创建虚拟环境
if [ ! -d "venv" ]; then
  echo "📦 创建 Python 虚拟环境..."
  python3 -m venv venv
else
  echo "✅ 虚拟环境已存在"
fi

# Step 2: 激活虚拟环境
echo "🔥 激活虚拟环境..."
source venv/bin/activate

# Step 3: 安装依赖
echo "📥 安装依赖中..."
pip install --upgrade pip
pip install fastapi uvicorn torch torchvision pillow numpy scikit-learn joblib tqdm

# Step 4: 启动 FastAPI 服务
echo "🚀 启动 FastAPI 服务中..."
uvicorn main:app --reload --host 0.0.0.0 --port 8000
